from torch import device , cuda
from os import makedirs
# Configuration
class Config:
    image_size = 32
    channels = 3  #RGB三通道
    batch_size = 64
    learning_rate = 1e-4
    timesteps = 1000
    device = device("cuda" if cuda.is_available() else "cpu")
    epochs = 1000
    num_samples = 1000  # number of images in your dataset
    data_dir = "32pic"
    save_dir = "saved_models_"+data_dir
    isplot=False
    seed=42
    makedirs(save_dir, exist_ok=True)